#-- coding: utf-8 --

'''
Author: 普拉斯777
Date: 2023年6月16日
Description: 截取“智慧考勤”月报表中需要的列
'''

import openpyxl
import os

def extract_data(input_file, output_file):
    wb = openpyxl.load_workbook(input_file)
    ws = wb.active

    # 提取前A、C、D、E列的数据
    data = [["姓名", "应出勤天数", "出勤天数（工作日）" ]]
    for row in ws.iter_rows(min_row=3, values_only=True):
        data.append([row[0], row[3], row[4]])

    # 将数据写入新的 Excel 文件
    output_dir = "./output"
    os.makedirs(output_dir, exist_ok=True)  # 如果目录不存在，就创建目录
    output_path = os.path.join(output_dir, output_file)

    wb = openpyxl.Workbook()
    ws = wb.active
    ws.title = "提取数据"
    for row in data:
        ws.append(row)
    wb.save(output_path)

    print(f"处理完成，结果已保存到文件：{output_path}")

def main():
    input_file = "yue.xlsx"
    output_file = "yue-cute.xlsx"

    extract_data(input_file, output_file)

if __name__ == "__main__":
    main()
